'''现在希望把之前案例中两个国家的数据方法一起来研究分析，
同时保留国家的信息（每条数据的国家来源），应该怎么办
'''

import numpy as np
from matplotlib import pyplot as plt


us_file_path = './youtube_video_data/US_video_data_numbers.csv'
uk_file_path = './youtube_video_data/GB_video_data_numbers.csv'

t_us = np.loadtxt(us_file_path,delimiter=',',dtype='int')
t_uk = np.loadtxt(uk_file_path,delimiter=',',dtype='int')


# t = np.stack(arrays=np.arange(t_us.shape[0]),axis=0)
# print(t)

# 添加国家信息
# 构造全为0的数据
zeros_data = np.zeros((t_us.shape[0],1)).astype(int)
ones_data = np.ones((t_uk.shape[0],1)).astype(int)

print(np.zeros)

# us_data = np.hstack((t_us,zeros_data))
# uk_data = np.hstack((t_uk,ones_data))
#
#
# # 评价两组数据
# final_data = np.vstack((us_data,uk_data))
# print(final_data)
